"""
解析和修改XML

问题：
    你想读取一个XML文档，对它最一些修改，然后将结果写回XML文档。
解决方案：
    使用 xml.etree.ElementTree 模块可以很容易的处理这些任务
"""

from xml.etree.ElementTree import parse, Element, tostring


def parse_modify_rewrite_xml():
    doc = parse("pred.xml")
    root = doc.getroot()
    root.remove(root.find('sri'))
    root.remove(root.find('cr'))
    print(root.getchildren().index(root.find('nm')))

    e = Element('spam')
    e.text = 'This is a test'
    root.insert(2, e)
    doc.write('new_pred.xml', xml_declaration=True)


if __name__ == '__main__':
    parse_modify_rewrite_xml()
